package cn.itsource.org.controller;

import cn.itsource.basic.util.AjaxResult;
import cn.itsource.basic.util.PageBean;
import cn.itsource.org.domin.Dept;
import cn.itsource.org.query.DeptQuery;
import cn.itsource.org.service.IDeptService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/*
Controller的升级：
    1、RESTful ： http://localhost:8080/dept/all【URL】 +  Http动词(GET\POST\DELETE\PUT\PATCH)【请求方式】
    2、请求参数为json ： 在参数列表参数前添加@RequestBody
                    前端测试时需要提交json格式的数据  axios默认提交的就是json
    3、添加修改合并
        通过提交的实体有没有id值判断要执行的业务
    4、接口的返回值
        dml的方法  增改   删    返回的是一个String 不是一个标准的json   返回值统一一下 使用一个类型来进行封装
        basic.util包中开发AjaxResult
 */
@Api(value = "部门的API",description = "部门相关的CRUD功能")
@RestController
@RequestMapping("/dept")
public class DeptController {

    @Autowired
    IDeptService service;

    /*添加修改合并*/
    //http://localhost:8080/dept + POST
    @PutMapping
    @ApiOperation(value = "添加或修改")
    public AjaxResult addOrUpdate(@RequestBody Dept dept) {

        AjaxResult successAjax = AjaxResult.me();
        //异常处理
        try {
            //判断没有id   就添加
            if(dept.getId()==null){
                //添加成功
                if(service.add(dept)){
                    //返回成功对象
                    return successAjax;
                }
                //返回失败对象
                return AjaxResult.me().setSuccess(false).setMessage("添加操作失败！！");
            }else{
                //判断修改成功
                if(service.update(dept)){
                    //返回成功对象
                    return successAjax;
                }
                //返回失败对象
                return AjaxResult.me().setSuccess(false).setMessage("修改操作失败！！");
            }
        } catch (Exception e) {
            e.printStackTrace();
            //发生异常返回失败对象 描述异常原因
            return AjaxResult.me().setSuccess(false).setMessage("添加或修改发生异常！！");
        }

    }

    //分页的方法实现
    @ApiOperation(value = "分页查询")
    @PostMapping  //http://localhost:8080/dept  + POST
    public PageBean<Dept> findByPage(@RequestBody DeptQuery query) {
        System.out.println(query);
        return  service.findByPage(query);
    }

    //http://localhost:8080/dept /13 +DELETE
    @DeleteMapping("/{id}")
    @ApiOperation(value = "删除方法")
    public AjaxResult del(@ApiParam(name="id",required = true)@PathVariable("id") Long id) {
        try {
            //判断是否删除成功
            if (service.del(id)) {
                return AjaxResult.me();
            } else {
                return AjaxResult.me().setSuccess(false).setMessage("删除操作失败！！请重试");
            }
        } catch (Exception e) {
            e.printStackTrace(); //发生异常返回失败对象 描述异常原因
            return AjaxResult.me().setSuccess(false).setMessage("删除异常！！请重试");
        }
    }
    /*批量删除 PatchMapping批量操作*/
    @PatchMapping //http://localhost:8080/dept  +Patch
    @ApiOperation(value = "批量删除方法")
    public AjaxResult batchRemove(@RequestBody List<Long> id){
        try {
            //判断是否删除成功
            if(service.batchRemove(id)){
                //成功返回成功对象
                return AjaxResult.me();
            }else {
                return AjaxResult.me().setSuccess(false).setMessage("批量删除操作失败！！请重试");
            }
        } catch (Exception e) {
            e.printStackTrace();
            return AjaxResult.me().setSuccess(false).setMessage("批量删除异常！！请重试");
        }
    }

    @GetMapping("/{id}")
    @ApiOperation(value = "通过id查询")
    //http://localhost:8080/dept/13 + GET
    public Dept findById(@ApiParam(name="id",required = true) @PathVariable("id") Long id) {

        return service.findById(id);
    }

    @GetMapping
    @ApiOperation(value = "查询所有")
    //http://localhost:8080/dept + GET    @GetMapping 请求这个方法时必须使用get方式
    public List<Dept> findAll() {
        return service.findAll();
    }

    @GetMapping("/pid/{id}")
    @ApiOperation(value = "通过parentId查询") //http://localhost:8080/dept/pid/13 + GET
    public List<Dept> findByParentId(@PathVariable("id") Long id){
        return service.findByParentId(id);
    }

     /*
       部门树结构封装的思路：
       1、查询所有部门  all
       2、转成一个map结构   key 部门id   value 部门对象
       3、遍历所有部门
           如果是一级部门  添加到最终返回的集合中
           如果有父级部门  找自己的父级部门 parent_id  将自己放入父级部门对象的children集合中
     */
    @ApiOperation(value = "无限级树")
    @GetMapping("/tree") //  http://localhost:8080/dept/tree + Get
    public List<Dept> tree(){
        //1.查询所有部门
        List<Dept> all = service.findAll();
        //2.转成map结构 key 部门id   value 部门对象
        Map<Long, Dept> map = new HashMap<>();
        //遍历all list 装到map中
        for (Dept dept:all) {
            map.put(dept.getId(),dept);
        }
        //3.遍历所有部门
        //如果是一级部门  添加到最终返回的集合中
        //如果有父级部门  找自己的父级部门 parent_id  将自己放入父级部门对象的children集合中
        //构建最终要返回的集合
        List<Dept> finalList = new ArrayList<>();
        for (Dept currentDept:all) {
            //判断当前部门是否是一级部门 即是否有parent_id
            if(currentDept.getParent_id()==null){
                //如果parent_id为空，则是一级部门 最终返回到集合中
                finalList.add(currentDept);
            }else{
                //如果不为空 则说明是子部门，找到父级部门的parent_id 将自己放入父级部门的对象的children集合中
               //获取父部门id
                Long pid = currentDept.getParent_id();
                //用父部门id在map中查找父部门对象
                Dept parentDept = map.get(pid);
                //将当前对象加入到父级部门的children集合中
                parentDept.getChildren().add(currentDept);
            }
        }
       return finalList;
    }


}
